package com.leyou.upload.service;

import com.github.tobato.fastdfs.domain.StorePath;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

@Service
public class UploadService {
    private Logger logger = LoggerFactory.getLogger(UploadService.class);
    private static Set<String> set = new HashSet<>();

    static {
        set.add("image/png");
        set.add("image/jpeg");
    }

    @Autowired
    FastFileStorageClient storageClient;


    public String upload(MultipartFile file) {
        /**
         *在上传文件过程中，我们需要对上传的内容进行校验：
         1. 校验文件大小
         2. 校验文件的媒体类型
         3. 校验文件的内容
         文件大小在Spring的配置文件中设置，因此已经会被校验，我们不用管。
         * */
        //支持的文件类型
        try {

            String contentType = file.getContentType();
            if (!set.contains(contentType)) {
                logger.info("文件类型不匹配:{}", contentType);
                return null;
            }
            //文件不能为空
            long size = file.getSize();
            if (size <= 0) {
                logger.info("文件内容为空，不符合要求");
                return null;
            }
            //图片内容
            BufferedImage image = ImageIO.read(file.getInputStream());
            if (image == null) {
                logger.info("内容不符合要求");
                return null;
            }
            //保存图片
            /*File dir = new File("D:\\Server");
            if (!dir.exists()) {
                dir.mkdirs();
            }
            file.transferTo(new File(dir, file.getOriginalFilename()));*/
            //获取文件后缀名
            String extension = StringUtils.substringAfter(file.getOriginalFilename(), ".");
            //上传
            StorePath storePath = storageClient.uploadFile(file.getInputStream(), file.getSize(), extension, null);

            //返回完整路径
            String url = "http://image.leyou.com/" +
                    storePath.getFullPath();
            return url;
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }

    }
}
